# import os
import time
from mindformers.tools.resume_ckpt import get_resume_checkpoint
from mindformers.tools.logger import logger

start_time = time.time()
resume_ckpt = get_resume_checkpoint("/data01/huawei/zsz/resume_training/checkpoint", True)
end_time = time.time() - start_time
#logger.info(os.path.exists("/data01/huawei/zsz/resume_training/checkpoint"))

logger.info(resume_ckpt)
logger.info(f"time: {round(end_time, 5)}s")
